package cn.enjoy.dynamic.datasource.config;

import java.util.ArrayDeque;
import java.util.Deque;

/**
 * 多数据源上下文
 *
 * @author K
 * @date 2020/12/14 15:41
 */
public class DynamicContextHolder {
    @SuppressWarnings("unchecked")
    private static final ThreadLocal<Deque<String>> CONTEXT_HOLDER = new ThreadLocal() {
        @Override
        protected Object initialValue() {
            // 数组队列 Object[16]
            return new ArrayDeque();
        }
    };

    /**
     * 获得当前线程数据源
     *
     * @return 数据源名称
     */
    public static String peek() {
        // 获取当前线程的值（队列），第一个元素
        return CONTEXT_HOLDER.get().peek();
    }

    /**
     * 设置当前线程数据源
     *
     * @param dataSource 数据源名称
     */
    public static void push(String dataSource) {
        // get获取当前线程的值，如果没有则初始化。 push Deque对象数组中放入值
        CONTEXT_HOLDER.get().push(dataSource);
    }

    /**
     * 清空当前线程数据源
     */
    public static void poll() {
        Deque<String> deque = CONTEXT_HOLDER.get();
        // 队列删除第一个元素（就是ThreadLocal get push初始化的值）
        deque.poll();
        if (deque.isEmpty()) {
            // 如果队列为空，就删除当前线程存储的值，后面如果有线程进来，又会初始化值
            CONTEXT_HOLDER.remove();
        }
    }

}
